# coding:utf-8
from record import Record
from dataset import Dataset


train_filename = 'data/train'
dataset = Dataset()
all_limit = 10000

with open(train_filename) as f:
    count = 0
    for line in f:
        line = line.split('\x01')
        record = Record(line)
        dataset.add(record)
        count += 1
        if count >= all_limit:
            break

dataset.output_label_1_attr('ua')
